/****************************************************************************
**
** WX公众号:Qt历险记
** CSDN:Qt魔术师
**
****************************************************************************/

#include "freezetablewidget.h"

#include <QScrollBar>
#include <QHeaderView>

//[构造函数]
FreezeTableWidget::FreezeTableWidget(QAbstractItemModel * model)
{
      this->setModel(model);
      frozenTableView = new QTableView(this);

      init();

      // 表头的大小调整时，会触发相应的槽函数，从而更新表格的宽度和高度
      connect(horizontalHeader(),&QHeaderView::sectionResized, this,
              &FreezeTableWidget::updateSectionWidth);
      connect(verticalHeader(),&QHeaderView::sectionResized, this,
              &FreezeTableWidget::updateSectionHeight);

      // 在滚动一个视图的垂直滚动条时，会同步更新另一个视图的垂直滚动条的值，从而实现两个视图的滚动同步。
      connect(frozenTableView->verticalScrollBar(), &QAbstractSlider::valueChanged,
              verticalScrollBar(), &QAbstractSlider::setValue);
      connect(verticalScrollBar(), &QAbstractSlider::valueChanged,
              frozenTableView->verticalScrollBar(), &QAbstractSlider::setValue);


}
//[END]

FreezeTableWidget::~FreezeTableWidget()
{
      delete frozenTableView;
}

//[初始化1]
void FreezeTableWidget::init()
{
      // 这行代码设置了冻结表格视图的模型（model）为当前的模型。
      frozenTableView->setModel(model());

      frozenTableView->setFocusPolicy(Qt::NoFocus); // 无焦点
      frozenTableView->verticalHeader()->hide();    // 垂直表头隐藏
      frozenTableView->horizontalHeader()->setSectionResizeMode(QHeaderView::Fixed); // 水平头大小模式：固定

      // 冻结表格视图置于当前视图的下方
      this->viewport()->stackUnder(frozenTableView);

      // 设置冻结表格视图的样式表，包括边框、背景颜色和选择时的背景颜色
      frozenTableView->setStyleSheet("QTableView { border: none;"
                                     "background-color: #8EDE21;"
                                     "selection-background-color: #999}");
      // 设置冻结表格视图的选择模型为当前视图的选择模型
      frozenTableView->setSelectionModel(selectionModel());
      // 循环代码隐藏了冻结表格视图中除第一列以外的所有列
      for (int col = 1; col < model()->columnCount(); ++col)
            frozenTableView->setColumnHidden(col, true);

      // 设置冻结表格视图的第一列宽度与当前视图的第一列宽度相同
      frozenTableView->setColumnWidth(0, columnWidth(0) );

      // 设置冻结表格视图的水平滚动条策略和垂直滚动条策略为始终关闭
      frozenTableView->setHorizontalScrollBarPolicy(Qt::ScrollBarAlwaysOff);
      frozenTableView->setVerticalScrollBarPolicy(Qt::ScrollBarAlwaysOff);
      frozenTableView->show();

      updateFrozenTableGeometry();

      // 分别设置当前视图和冻结表格视图的水平和垂直滚动模式为每次滚动一个像素
      this->setHorizontalScrollMode(ScrollPerPixel);
      this->setVerticalScrollMode(ScrollPerPixel);
      frozenTableView->setVerticalScrollMode(ScrollPerPixel);
}


// 第一列宽度[槽]
void FreezeTableWidget::updateSectionWidth(int logicalIndex, int /* oldSize */, int newSize)
{
      if (logicalIndex == 0){
            frozenTableView->setColumnWidth(0, newSize);
            updateFrozenTableGeometry();
      }
}

// 行高[槽]
void FreezeTableWidget::updateSectionHeight(int logicalIndex, int /* oldSize */, int newSize)
{
      frozenTableView->setRowHeight(logicalIndex, newSize);
}
//[END]


//[resizeEvent函数用于处理窗口大小改变的事件。首先，它调用了QTableView类的resizeEvent函数，以确保父类的resizeEvent函数也能被执行。
// 然后，它调用了updateFrozenTableGeometry函数，用于更新冻结表格的几何位置]
void FreezeTableWidget::resizeEvent(QResizeEvent * event)
{
      QTableView::resizeEvent(event);
      updateFrozenTableGeometry();
 }
//[END]


/*
该函数的作用是在表格视图中移动光标，并在需要时滚动视图以保持冻结列的固定位置。
函数首先调用基类QTableView的moveCursor()函数来移动光标，并将返回的当前索引赋值给current变量。
然后，函数检查光标移动的动作cursorAction是否为MoveLeft（向左移动）。
如果是向左移动的操作，并且当前索引的列号大于0，并且当前索引的左上角坐标的x值小于冻结表格视图（frozenTableView）第一列的宽度，那么就需要进行滚动操作。
滚动操作通过计算滚动条的新值来实现，新值的计算基于以下公式：
const int newValue = horizontalScrollBar()->value() + visualRect(current).topLeft().x()
                    - frozenTableView->columnWidth(0);
其中，horizontalScrollBar()->value()获取当前水平滚动条的值，visualRect(current).topLeft().x()获取当前索引单元格的左上角在视图中的x坐标，
frozenTableView->columnWidth(0)获取冻结表格视图第一列的宽度。
最后，通过horizontalScrollBar()->setValue(newValue)将滚动条的值设置为新值，实现滚动操作。
通过这样的实现，可以在向左移动光标时，如果光标位于冻结列的左侧，则滚动视图，使得当前光标所在的列仍然可见，同时保持冻结列的位置不变。
*/
QModelIndex FreezeTableWidget::moveCursor(CursorAction cursorAction,
                                          Qt::KeyboardModifiers modifiers)
{
      QModelIndex current = QTableView::moveCursor(cursorAction, modifiers);

      if (cursorAction == MoveLeft && current.column() > 0
              && this->visualRect(current).topLeft().x() < frozenTableView->columnWidth(0) ){

            const int newValue = this->horizontalScrollBar()->value() + this->visualRect(current).topLeft().x()
                                 - frozenTableView->columnWidth(0);
            this->horizontalScrollBar()->setValue(newValue);
      }
      return current;
}
//[END]

// 滚动到指定的索引位置
void FreezeTableWidget::scrollTo (const QModelIndex & index, ScrollHint hint){
    if (index.column() > 0)
        QTableView::scrollTo(index, hint);
}

// 该函数使用setGeometry()方法设置了冻结表格视图的位置和大小。冻结表格视图的横坐标为垂直表头的宽度加上视图的边框宽度，
// 纵坐标为视图的边框宽度，宽度为当前视图第一列的宽度，高度为视口的高度加上水平表头的高度。
// 通过调用这个函数，可以确保冻结表格视图与当前视图的几何形状保持一致，从而实现它们之间的同步显示。]
void FreezeTableWidget::updateFrozenTableGeometry()
{
      frozenTableView->setGeometry(this->verticalHeader()->width() + this->frameWidth(),
                                   this->frameWidth(), columnWidth(0),
                                   this->viewport()->height()+this->horizontalHeader()->height());
}
//[END]


